#!/bin/sh
python predict.py \
--dataset vaihingen \
--model /media/allen/backup/毕业论文实验结果/09/09_best.pth \
--root-dir /home/allen/mm_dataset/vaihingen_600_normal \
--output-dir /media/allen/backup/毕业论文实验结果/09/small_img \
--nclasses 6 \
--gpu-id 0